package dp;

/**
 * 122. 买卖股票的最佳时机2
 * 给定一个数组 prices ，它的第i 个元素prices[i] 表示一支给定股票第 i 天的价格。
 * 设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易（多次买卖一支股票）。
 * 注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
 *
 */

public class BestTimeToSellAndBuyStock_122 {
    // 由于可以在同一天先卖后买，所以只要是有利润就买卖即可
    public int maxProfit2(int[] prices) {
        int maxProfit = 0;
        for (int i = 0; i < prices.length - 1; i++) {
            if (prices[i] < prices[i + 1]) {
                maxProfit += prices[i + 1] - prices[i];
            }
        }
        return maxProfit;
    }

    // 由于可以在同一天先卖后买，所以只要是有利润就买卖即可
    public int maxProfit(int[] prices) {
        // 每天都有持有股票和卖出股票的2种状态
        // 定义动态数组dp[i][0]表示第i天未持有股票的最大收益，dp[i][1]表示第i天持有股票的最大收益

        int[][] dp = new int[prices.length][2];
        // 定义base case
        dp[0][0] = 0;
        dp[0][1] = -prices[0];

        for (int i = 1; i < prices.length; i++) {
            // 如果在第i天未持有股票，说明可能是在第i-1天也没有持有股票，或者在第i天卖了股票
            dp[i][0] = Math.max(dp[i-1][0], dp[i-1][1]+prices[i]);

            // 如果在第i天持有股票，说明可能在第i-1天持有股票，或者在第i天买入股票
            dp[i][1] = Math.max(dp[i-1][1], dp[i-1][0]-prices[i]);
        }
        return Math.max(dp[dp.length-1][0], dp[dp.length-1][1]);
    }
}
